package com.xj.algorithm.leetcode;

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;

/**
 * 两个数组的交集II
 */
public class L350_两个数组的交集和次数 {

    /**
     * 给定两个数组，编写一个函数来计算它们的交集。
     * <p>
     * 示例 1:
     * <p>
     * 输入: nums1 = [1,2,2,1], nums2 = [2,2]
     * 输出: [2,2]
     * 示例 2:
     * <p>
     * 输入: nums1 = [4,9,5], nums2 = [9,4,9,8,4]
     * 输出: [4,9]
     * 说明：
     * <p>
     * 输出结果中每个元素出现的次数，应与元素在两个数组中出现的次数一致。
     * 我们可以不考虑输出结果的顺序。
     * 进阶:
     * <p>
     * 如果给定的数组已经排好序呢？你将如何优化你的算法？
     * 如果 nums1 的大小比 nums2 小很多，哪种方法更优？
     * 如果 nums2 的元素存储在磁盘上，磁盘内存是有限的，并且你不能一次加载所有的元素到内存中，你该怎么办？
     */
    public static void main(String[] args) {
        int[] nums1 = new int[]{1, 2, 2, 1};
        int[] nums2 = new int[]{2, 2};

        int[] res = intersectOffice(nums1, nums2);
        for (int val : res) {
            System.out.print(val + ",");
        }
        System.out.println();

        int[] nums3 = new int[]{4, 9, 5};
        int[] nums4 = new int[]{9, 4, 9, 8, 4};

        int[] res2 = intersectOffice(nums3, nums4);
        for (int val : res2) {
            System.out.print(val + ",");
        }
        System.out.println();
    }

    //自己的解法；hashmap找相同的结果,统计每个值出现的次数
    public static int[] intersect(int[] nums1, int[] nums2) {
        Map<Integer, Integer> map = new HashMap<>();
        for (int value : nums1) {
            Integer count = map.get(value);
            if (count == null) {
                map.put(value, 1);
            } else {
                map.put(value, ++count);
            }
        }

        int[] result = new int[map.size()];
        int index = 0;
        for (int value : nums2) {
            Integer count = map.get(value);
            if (count != null && count>0) {
                result[index++] = value;
                count--;
                map.put(value,count);
            }
        }

        int[] value = new int[index];
        for (int i = 0; i < index; i++) {
            value[i] = result[i];
        }
        return value;
    }

    //官方解法；排序，然后双指针
    public static int[] intersectOffice(int[] nums1, int[] nums2) {
        Arrays.sort(nums1);
        Arrays.sort(nums2);
        int i = 0, j = 0, k = 0;
        while (i < nums1.length && j < nums2.length) {
            if (nums1[i] < nums2[j]) {
                ++i;
            } else if (nums1[i] > nums2[j]) {
                ++j;
            } else {
                nums1[k++] = nums1[i++];
                ++j;
            }
        }
        return Arrays.copyOfRange(nums1, 0, k);
    }

}
